import segmentation_models_pytorch as smp
model_unet_res50 = smp.Unet(
    encoder_name="resnet50",  # choose encoder, e.g. mobilenet_v2 or efficientnet-b7
    encoder_weights="None",  # use `imagenet` pretreined weights for encoder initialization
    in_channels=3,  # model input channels (1 for grayscale images, 3 for RGB, etc.)
    classes=10,  # model output channels (number of classes in your dataset)
)